多 Token 预测的正确优化法:Soft-MTP
🧭 LLM 预训练 · Multi-Token Prediction
MIT
提出 Soft-MTP,通过一行代码改动移除标准 MTP 训练中的噪声残差梯度,只保留 KL 自蒸馏信号,全面提升性能与推理加速
📌 工作参考价值(3 点)
- 若内部训练 LLM 用于生成式推荐(如商品描述、搜索 query 推荐、对话式导购),MTP 是一个可落地的训练技巧。采用 Soft-MTP 的一行改动可同时提升模型困惑度和下游准确率,且无额外计算开销,适合直接加到现有预训练流程。
- 在需要 speculative decoding 加速在线推理的场景(如实时搜索建议、Agent 规划生成),Soft-MTP 训练出的 MTP head 能提供更准确的 draft tokens,显著提升 decoding 接受率和加速比,降低延迟。
- 该 KL gradient-recovery trick 可复用到其他共享输出头的自蒸馏任务,例如知识蒸馏或辅助损失设计,避免直接计算大词表 KL 导致的内存瓶颈,兼容现有 fused cross-entropy kernels,工程实现简单。
多Token预测与替代分词改进大规模隐式多跳图搜索学习
🧭 多Token预测与分词改进多跳推理学习
University of Toronto · Vector Institute · Manulife Applied AI Research · Dalhousie University
用多token预测和替代分词突破LLM学习隐式多跳图搜索的规模墙,实现稳定训练
📌 工作参考价值(4 点)
- MTP 可作为辅助训练目标加入生成式推荐/Query 改写模型:若任务需要隐式多步推理(如从用户意图推导 query、从 query 推导商品路径),同时预测未来多个 token 或路径节点,能强化模型对中间步骤的规划能力。
- 分词策略对结构化生成任务影响显著:将图路径、商品序列或意图链按更合适的语义单元编码(如子路径、节点对、结构化 token)替代逐节点编码,可降低序列长度、改善学习稳定性,在电商路径生成中值得单独调优。
- 训练信号与分词是两个正交因素,可以独立迭代;工程实现上可先在现有生成式推荐模型上分别尝试 MTP 和调整 item/path tokenization,无需整体重构。
- 论文使用合成图任务,主要是学术贡献,业务直接迁移有限,但其“扩展墙源于监督信号与分词”的归因可迁移到其他需要长程依赖的生成场景(如会话内多跳推荐、广告序列生成)。
EchoRec:基于循环一致偏好对齐的多Token预测生成式推荐
🧭 生成式推荐 · Multi-Token Prediction ★ 深度解读
National University of Singapore · Beijing University of Posts and Telecommunications · Tencent · Shandong University
用未来多步行为作顺序依赖的密集监督,并通过循环一致性对齐把整体偏好内化到解码表示,提升生成式推荐且在线零开销
📌 工作参考价值(4 点)
- 未来行为可作为辅助监督:在训练时挂轻量 MTP 分支预测 t+2、t+3 商品,并让每个分支条件在前一个分支表示/上一步商品上,而不是从同一隐状态并行解码;能避免多个 horizon 预测同质化,适合需要一次性生成多个候选、多创意、多坑位结果的广告/推荐场景。
- 训练时新增的辅助分支共享主模型输出头、token embedding 和 decoding graph,推理时全部丢弃,线上延迟与显存几乎不变。这适合电商/广告高 QPS 场景:用可接受的训练成本换在线零开销的指标提升。
- 若业务中需要对齐多个表示(如用户向量与多意图向量、跨模态特征、多任务共享表示),单向 alignment 可能被 projector 吸收差异,造成“虚假对齐”;加反向 projector 和 round-trip/cycle loss 可抑制这类问题,是一个可复用的训练 trick。
- 当用户行为序列较长且噪声较多时,把目标从单点 next-item 改为多 horizon 整体偏好,会提升对噪声交互的鲁棒性;论文在 20% 随机替换噪声下相对提升反而从 17.68% 增至 54.89%,对真实电商点击噪声有参考价值。
AngelSpec: 面向真实负载的推测解码训练与部署框架
🧭 推测解码 · 多策略协同与自适应验证 ★ 深度解读
Tencent Inc.
针对负载异构,专业化MTP与块扩散drafter,配合自适应验证深度,在Hy3-21B上实现1.98–2.40倍加速
📌 工作参考价值(4 点)
- 若业务LLM服务需加速(如推荐解释生成、query改写),可借鉴“工作负载特化”思想:为闲聊等高熵场景训练MTP drafter;为代码生成等低熵场景训练块扩散drafter,提升接受率。
- D-cut的自适应验证深度策略可迁移至Agent多步推理:根据请求置信度与系统负载动态决定LLM生成步数,平衡响应时间与质量,避免固定budget浪费。
- 训练技巧:Training-Time Test(TTT)和基于总变差距离(TV loss)的损失函数可有效缓解drafter分布偏移,对低延迟场景(如在线推荐)有直接应用价值,可复用其接受率对齐训练范式。
- AngelSpec开源训练框架支持多drafter联合训练与评估,降低企业自研推测解码方案的门槛,适合多领域LLM应用的快速实验。
HGenPush:面向工业推送通知系统的异构生成式推荐架构
🧭 生成式推荐 · 异构内容联合生成 ★ 深度解读
Kuaishou Technology
解耦自回归生成,以链式多token预测和行为偏好对齐统一推荐视频与作者,快手推送获+0.181% DAU提升。
📌 工作参考价值(4 点)
- 链式多token预测(Chained-MTP)替代自回归:用稳定的用户兴趣表示作为全局上下文,叠加前序语义ID的累计嵌入来建模层级依赖,整个序列的token可并行预测。相比Transformer级联方式,推理吞吐提升34%,效果基本持平,很适合需要高QPS的生成式推荐落地。
- 异构目标统一生成:对作者(或店铺、媒体源)采用行为偏好对齐得到偏好感知的embedding,经RQ-Kmeans离散化为语义ID,再与视频一级语义ID拼接成混合ID,用同一个decoder生成。这样在系统中自然融合商品推荐与信任源推荐,可直接借鉴到电商中“商品+店铺”或“文章+作者”的组合推荐。
- 用户消费偏好对齐(UCPA + GSISPO):用点击后会话中真实消费行为(播放完成、点赞、转发等)作为奖励信号,序列级计算重要性采样权重并裁剪,避免token级优化的语义碎片化,同时保留高价值长尾行为的梯度。该RL对齐方案无需额外critic网络,能有效抑制标题党,提升深度消费指标。
- 多场景多视角行为融合:将长/短期feed行为、推送点击行为、推送发送行为与用户静态特征串联输入causal decoder,分别抽取[cls]表示供不同生成分支使用。这种跨场景行为组合及特殊token设计,可迁移到需要跨业务线联合建模的用户理解模块。
Nemotron 3 Ultra:面向 Agent 推理的 550B 混合专家 Mamba-Attention 开源模型
🧭 Agent 专用大模型 · 混合 Mamba-Attention MoE ★ 深度解读
NVIDIA
总参数 550B、活跃 55B 的混合 Mamba-Attention MoE 模型,在 Agent 推理上精度对齐且推理吞吐量最高达同规模模型的 5.9 倍
📌 工作参考价值(4 点)
- 长上下文高效推理架构可迁移到推荐 Agent 中:Hybrid Mamba‑Attention + MoE 大幅降低 KV cache 与注意力成本,适合需要维持超长对话历史或文档上下文的推荐/搜索 Agent;可在电商客服、导购助手等场景落地类似模式,平衡推理延迟与能力。
- Multi‑Token Prediction 头可直接加速自回归生成:训练时加入共享权重的 MTP 头,上线后通过投机解码(speculative decoding)提升吞吐,对需要高并发生成回复的推荐理由、搜索词改写等场景有直接工程价值。
- 后训练中的多教师 On‑Policy 蒸馏(MOPD)可借鉴到工业 Agent 对齐:用领域专家 teacher 在 student 的 rollouts 上进行 token 级蒸馏,能在保持推理效率的同时注入多任务 Agent 能力(如工具使用、纠错),适合业务定制化 Agent 构建。
- 推理预算控制(reasoning effort control)可用于动态服务质量调节:允许在推理时根据负载动态调整思考深度,平衡效果与成本,可应用于推荐 Agent 的实时适配。
打破熵的限制:多token预测与拒绝采样加速强化学习训练
🧭 LLM强化学习训练加速 · 多token预测
Qwen Team, Alibaba Inc.
揭示RL阶段熵升高与MTP接受率的负线性关系,通过拒绝采样与TV损失将接受率提至95%,实现1.8倍训练加速
📌 工作参考价值(4 点)
- 拒绝采样替代贪婪采样:在电商对话式推荐或Agent策略生成中,当使用MTP(或多token生成)加速推理时,若模型因在线学习(如RL更新)导致输出分布偏移,应使用概率拒绝采样而非贪婪接受,以保持高接受率和加速比。
- 端到端优化接受率指标:推荐或生成模型中若采用多步草稿验证机制(如Sequential Recommendation中的beam search加速),可设计类似TV loss的目标函数,直接最大化多步联合接受率,替代中间步骤的KL或交叉熵损失,可提升整体推理吞吐。
- 预训练加速模块避免在线开销:在RL训练的rollout阶段,MTP模块可在RL开始前用端到端损失充分训练并冻结,避免在线更新引入的计算代价和分布波动,这对需要稳定服务的推荐Agent尤其有用。
- 熵监控指导加速策略:实际部署中,可监测生成熵值动态调整MTP步长或是否启用加速,当熵上升时自动切换到拒绝采样或回退到逐token生成,平衡效率与质量。
Pair-In, Pair-Out:统一输入压缩与输出多token预测的高效LLM推理
🧭 高效推理 · 输入压缩与多token预测统一 ★ 深度解读
Gaoling School of Artificial Intelligence, Renmin University of China · AI Platform, Xiaohongshu Inc. · University of Electronic Science and Technology of China · Institute of Automation, Chinese Academy of Sciences
通过对称的 pair-in 压缩和 pair-out MTP 头,用 on-policy 蒸馏免费训练的置信头替代验证器,实现精度不降的推理加速。
📌 工作参考价值(4 点)
- 将输入压缩与输出多token预测视为对称操作:若业务下游需部署推理流水线,可借鉴 PIPO 将 compressor 和 MTP head 镜像组合,降低工程复杂度,且可直接利用已有 MTP 模块。
- 用 on-policy 蒸馏的教师分布训练置信头,无需额外成本:在已有 OPD 流程中,教师/学生 token 概率天然提供 speculative decoding 的接受率标签,可零开销训练一个轻量 MLP 替代昂贵的 verifier 前传,适合电商/Agent 场景中高频推理的成本控制。
- 通过随机 padding 注入训练,让模型学会单双token灵活切换:业务中可能遇到不确定的 draft 质量,可参照 PIPO 的做法,在 SFT 阶段按比例插入 padding token,使模型获得拒绝 draft 的回退能力,提升可靠性。
- 在长上下文高吞吐场景优势明显:PIPO 的 TTFT 加速在长 prompt 时更显著(2K→2.64× @128K),适合需要处理长文档、多轮对话或复杂推理链的电商推荐解释、Agent 规划等任务。
MiniMax-M2 系列:以小激活释放最大真实世界智能
🧭 MoE 架构与 Agent RL 训练协同设计 ★ 深度解读
MiniMax
以仅 9.8B 激活参数的 MoE 架构,通过 agent 原生数据管道和 RL 系统达到 agentic 前沿性能,并初步实现自进化。
📌 工作参考价值(4 点)
- 稀疏 MoE 设计:256 细粒度专家,sigmoid gating 和可学习偏置大幅降低辅助损失,每 token 仅激活 8 个专家。在电商推荐中可用于生成式模型,兼顾大容量与低推理成本,其负载均衡策略可迁移到多任务学习。
- Agent RL 训练系统 Forge:训练-推理-agent 解耦,支持白盒/黑盒 agent,用窗口化 FIFO 和前缀树合并吸收长短轨迹差异。可直接复用到多智能体推荐系统的强化学习组件,提升离线策略优化效率。
- 可验证轨迹数据合成:用 GitHub PR 构造 SWE 任务,Docker + 测试用例提供硬奖励;AppDev 采用专家 meta query + Agent-as-a-Verifier 三层验证。推荐系统可借鉴“可执行验证”思想,为商品文案生成等任务自动合成带奖励的高质量数据。
- 自进化能力:M2.7 能自主调试训练、修改自身 scaffold,展示从训练到部署的自动化闭环。可用于推荐模型的在线自动纠错、特征自动更新等运维场景,减少人工介入。